package levelorder;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

public class LevelTraversalOfNTree {
    /*
     * 429.N叉树的层序遍历
     * 给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。
     * 树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。
     *
     * 树的高度不会超过 1000
     * 树的节点总数在 [0, 10^4] 之间
     *
     * */
    public static void main(String[] args) {

    }

    public List<List<Integer>> mySolution(Node root) {
        ArrayList<List<Integer>> res = new ArrayList<>();
        if(root == null)
            return res;
        LinkedList<Node> deque = new LinkedList<>();
        deque.add(root);
        while (!deque.isEmpty()){
            int len = deque.size();
            ArrayList<Integer> r = new ArrayList<>();
            while (len > 0){
                Node node = deque.pop();
                r.add(node.val);
                if(node.children != null || node.children.size() != 0){
                    for(Node item : node.children){
                        deque.add(item);
                    }
                }
                len--;
            }
            res.add(r);
        }
        return res;
    }

}

class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int _val) {
        val = _val;
    }

    public Node(int _val, List<Node> _children) {
        val = _val;
        children = _children;
    }
}
